package cn.com.mall.common.valid;

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

/**
 * 自定义校验器
 */
public class ListValidConstraintValidator implements ConstraintValidator<ListValue,Integer> {

    Set<Integer> valueSet=new HashSet<>();

    @Override
    public void initialize(ListValue constraintAnnotation) {

       int[] values=constraintAnnotation.values();
       if(values!=null){
           Arrays.stream(values).forEach((item)->{
               valueSet.add(item);
           });
       }

    }

    @Override
    public boolean isValid(Integer value, ConstraintValidatorContext context) {
        return valueSet.contains(value);
    }
}
